from .partA2_head import PartA2FCHead
from .pointrcnn_head import PointRCNNHead
from .pvrcnn_head import PVRCNNHead
from .second_head import SECONDHead
from .voxelrcnn_head import VoxelRCNNHead
from .voxelrcnn_kl_label_head import VoxelRCNNKLLabelHead
from .voxelrcnn_kl_label_iou_head import VoxelRCNNKLLabelIoUHead, VoxelRCNNKLLabelIoUPleHead

from .roi_head_template import RoIHeadTemplate


__all__ = {
    'RoIHeadTemplate': RoIHeadTemplate,
    'PartA2FCHead': PartA2FCHead,
    'PVRCNNHead': PVRCNNHead,
    'SECONDHead': SECONDHead,
    'PointRCNNHead': PointRCNNHead,
    'VoxelRCNNHead': VoxelRCNNHead,
    'VoxelRCNNKLLabelHead': VoxelRCNNKLLabelHead,
    'VoxelRCNNKLLabelIoUHead': VoxelRCNNKLLabelIoUHead,
    'VoxelRCNNKLLabelIoUPleHead': VoxelRCNNKLLabelIoUPleHead
}
